package com.example.niuke;

import java.util.ArrayList;
import java.util.Arrays;

/**
 * 牛客-数组中相加和为0的三元组
 * 描述
 * 给出一个有n个元素的数组S，S中是否有元素a,b,c满足a+b+c=0？找出数组S中所有满足条件的三元组。
 * 注意：
 * 三元组（a、b、c）中的元素必须按非降序排列。（即a≤b≤c）
 * 解集中不能包含重复的三元组。
 * 例如，给定的数组 S = {-10 0 10 20 -10 -40},解集为(-10, -10, 20),(-10, 0, 10)
 * 0 <= S.length <= 1000
 * 示例1
 * 输入：
 * [0]
 * 返回值：
 * []
 * 示例2
 * 输入：
 * [-2,0,1,1,2]
 * 返回值：
 * [[-2,0,2],[-2,1,1]]
 */
public class TripleAddsto0inanarray {
    public static void main(String[] args) {
        TripleAddsto0inanarray tripleAddsto0inanarray = new TripleAddsto0inanarray();
        int[] a = {1, 3, 2, 5, -2, -1, -3, 0};
        System.out.println(tripleAddsto0inanarray.threeSum(a));
    }

    public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if (num == null || num.length < 3) {
            return res;
        }
        Arrays.sort(num);
        for (int i = 0; i < num.length; i++) {
            if (num[i] > 0) {
                break;
            }

            int first = num[i];
            int left = i + 1;
            int right = num.length - 1;
            if (i > 0 && num[i] == num[i - 1]) {
                continue;
            }
            int[] ints = Arrays.copyOfRange(num, 0, 1);
            while (left < right) {
                int sum = first + num[left] + num[right];
                if (sum == 0) {
                    ArrayList<Integer> list = new ArrayList<>();
                    list.add(first);
                    list.add(num[left]);
                    list.add(num[right]);
                    res.add(list);

                    while (left < right && num[left] == num[left + 1]) {
                        left++;
                    }
                    while (left < right && num[right] == num[right - 1]) {
                        right--;
                    }
                    left++;
                    right--;
                } else if (sum > 0) {
                    right--;
                } else if (sum < 0) {
                    left++;
                }
            }
        }
        return res;
    }
}

